// Copyright (c) Appeon. All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. 
// You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0.

using System.Collections.Generic;
using System.Linq;
using System.Text;

namespace SnapObjects.Data.Oracle
{
    internal class OracleFunctionBuilderParser : FunctionBuilderParser
    {
        public OracleFunctionBuilderParser(
            ISqlStatementFactory factory,
            ISqlGenerationHelper helper,
            ITypeMapper typeMapper) : base(factory, helper, typeMapper)
        {
        }

        protected override IEnumerable<SqlToken> ParseTokens(string rawString)
        {
            using (ISqlTokenizer tokenizer = new OracleSqlTokenizer(rawString, _helper))
            {
                return tokenizer.ToList();
            }
        }

        public override SqlStatement Parse(SqlBuildStatement statement)
        {
            var sqlStatement = base.Parse(statement);

            if (sqlStatement is SqlFunctionStatement functionStatement)
            {
                foreach (var parameter in statement.Parameters)
                {
                    if (functionStatement.Parameters.TryGetValue(parameter.Name, out var v))
                    {
                        v.Direction = parameter.Direction;
                    }
                }
            }

            return sqlStatement;
        }
    }
}
